import torch
print(torch.cuda.is_available())  # True则说明GPU可用
print(torch.cuda.current_device())
print(torch.cuda.get_device_name(0))
